Measuring code performance - Python