Setup classes and initialized them. Some code clean up.