Added notes to assignment1b and cleaned code