remove code & add header to the controller