Added the time out class