Remove assignment 4 part 2

Missed out some stuff