Working on array exercises

Need to fix a bug