ENH: added tz support