Python: Faster Way