Cleaned up docs a bit, added more docs