Working on adding documentation