Move HeartBeat class to main package