small fix
minor updates