Cleaning up code -- removed unnecessary lines, fixed indentation, renamed method.