Adding minor features and documentation