Finished Wk2 assignment