Added some more comments, fixed print statement