Slight assignment4 updates