Updates for first homework assignment