Version 2

Restart functionality added. Added another class.