Improve documentaton