Add the bomb and robot function, and refactor the codes.