First commit for wk2 assignment