Improved task 1 code