Adding controller part