Final changes from Wk4 coding lesson