Changed code to use controller