Added display_array function