Fixed code for level 1