Fixed minor bugs, added in a comment, ready for HW4_1