Adding ArrayList for better testing