I'm still learning how to train models properly, but I'm hoping this one will start giving me better results