add controller back