Cleaned up and fixed documentation