Add tests

Maybe include the interactive prompt?