Changed code for level 2