Second commit, added the class