Pushing code for the 2nd assignment