fix formatting
improved unit test