Short assignment 2