Done with assignment 6