python code optimized