Add JavaDoc, minor fixes