Having problems passing info from one function to another (Python)