Fixed unit tests
Fixed unit tests

FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject,FrikkieSnyman/COS301_GroupProject