Improve documentaiton