Module: (Python) Conditional statement