Fixed up Classification

Fixed up the classification algorithm. Based on the sheet from work and
not online source. Done to comply with in company standards.
TestUnit

Added a JUnit