Clean up a few lines of code

Made the program do what I felt comfortable with turning in.