Finished screen implementation
Created tests for all cases and wrote java doc