Unit tests

#29