Small fixes and documentation