Submitted code for level 2b