assignment-1 -- DONE