Add tests
Fixes return syntax to [] => error/success