Minor updates on docs and utility functions