Change add_cap to capacity in image