Unit tests

Setup unit tests ready to fill out