Bug fixed for project 1