Added a bit more creative test
Added doc comments