fix remove student bug, minor improvements