Feature Extraction in Time-Frequency Domain - Audio Classification

Hi, I'm fairly new to MATLAB, really appreciate if I can get some help here.

I'm doing a project on audio classification, where I have loaded labelled audio datasets into MATLAB. I am currently encountering an issue where I want to apply discrete wavelet transform on all the signals in the datasets and convert them to images, subsequently saving them in a folder according to their labels. The images will then be fed into a neural network for classification learning.

I am trying to replicate another projet where they created a function to automatically convert CWT coefficients into images but I'm having no luck trying to make it work for DWT.

Thanks in advance for helping, and if I'm missing out any details please let me know.

my code:

https://pastebin.com/nCxzFqeN

The function code for CWT

https://pastebin.com/P6bZG6T7