Possible fix to time problem using pytz library