Got arraylist working