Start implementing unit tests