Javadoc + clean up