Added tests for different output
Made tests pass
Updated versions