Re Replace Python Function Return