Completed code to end Level 4