Cleaning up sketch 3