Put ArrayList and data in Data class
Adding methods
Adding more methods
Assignment-ready version