still trying to get the arraylist working