Small update to library