Minor changes in images controller