Done with assignment one