Calculate factorial in Python