Style fixes and documentation