added docs, removed unnecessary classes