Setup a nice python REPL