Minor changes (bug fixes)