Adding tests and comments to class files