Move controller in the directive