Little more assignment 3 progress