Started with cleaning up codebase, added some docs