Assignment 5a updates