Task5 lab2 test

Unit tests are out of date. Updates required