Clean up of classes.

Signed-off-by: Dimitrios Mandekis <mandekisd@gmail.com>