#4 Working on documentation