added operator to print out based on correct choice