Got mostly all stuff working sans class management. Need to write a directive for it