Cleaning up sketch 9