cleaning up code

cleaned up the few lines and added some output to indicate it's working