Working on improving the documentation