simplified code, still should work at this pt
adding more complicated cases to arrays