fix the code - code is not working