Apple will focus on improving battery capacity for next-gen iPhones