How can I know how to improve code performance in order to dont get timeout (python)? Can someone help?