Cleaned up files and added more docs