Python How To Split Code