main class added, add methods added, not everything works