Added editing feature for galeries too