Add documentation & code style fixes