Fixed missing ArrayList import