Done with assignment 1