add some python shell magic