Update class, create methods on validation of data. Start creating
correct JUnit test