Unit testing

Built a testing suite for the project