Extra credit for homework 2
cleaning up graphics