Update ug?

I added a new if statement.