Work on docs and fix build