Cleaning up sketch 5