minor changes to the lib