Some cleanup

1. Moved some asserts to earlier in the code
2. Using ```pytz.utc.zone``` instead of ```"UTC"```