Added the ability to add annotations to classes
Applied tests to both classes and methods