Cleanup

+ Added comments
+ Moved fuctionality of displaying time to seperate function
+ Little cleanup of code