Added UI elements for penalties