Add more tests for handling time zones