Added controller extension