Is OTHER wrong in the post ? POST : AITA for talking back to a superior?

I work at McDonald's and I am 16, meaning it's $13.50/hr (AUD).

I've been working there 3 months meaning I am now an employee and no longer a trainee, but I was rostered with a crew trainer making her my Superior (we will call her Joane). She is normally a very mean person and me and her were on 'OK' terms. 

This shift I was with her and it was alright but she was being mean to me like always, and ever since I was a trainee I would be a pushoverand just say "ok" and take it. But at one point she said "I expect more from trainee of the month". This made me slightly angry because she had voted me for that acheivement, but now she was throwing it back at my face, so I replied back with "It isn't that hard to get, you just gotta work a sh*tload of shifts" She then started going off at me about my attitude, and I assume she was taken back by me talking back. 
Later she started talking to the manager about my attitude which I called her out for and she finally stopped. We were very quiet for basically the rest of the shift.

Then when my shift was supposed to end I still had more to do, that was vital for the people closing. As I was doing it, I was still managing helping in kitchen because Joane couldn't handle it. She thought that I was purposefully taking longer to get more money, so she asked me for my code, I was very hesitant but I eventually gave it, and worked for 20mins without pay, and everyone including the manager saw it happen.
Once I had finished my shift I went to change and I saw Joan not working but instead with a drink at the back (there were other people in kitchen at this point).

After changing I apologised to Joane because I genuinely felt bad, I didn't realise I had upset her so much, just for talking back once. 

AITA? I feel really bad about it. I guess I should've treated my Superior with authority? But then again she doesn't treat any other crew members with that attitude.