Add a better REPL