First running version of the assignment 2