added my own unit test