add comments to classes
write a readme