Finished hw01 with comments