Update project 1 code. Only lacking While statement