added unit test to check if everything works