Write unit tests

Created by: Flabelatus