Add time module and make everything work