Modified unit tests  to test for correct behaviour when user submits wrong solution