changes code to match solution