Almost there, need to fix and add in lists array.