add simple class and change public to protected methods