Fixed image problem - add controller