fix teh bugs in level 4