Write proper docstring for each file, add to readme