Fixat och optimerat klasser

Fixat klasser och uppdaterad javadoc för detta.