Added option to make image square or single row.