If was stated all along that track limits wouldn't be enforced on Turn 4 during the race, only in practice and qualifying, so he wasn't doing anything wrong, but Marshalls decided to complain about it after Red Bull explicitly told Max what was happening and he started to use more of Turn 4.
The white lines should be enforced for the whole race as far as I am concerned. It was good to see Max erring on the side of caution and giving the place back with no histrionics, though given the above I do feel he was a little hard done by in the circumstances.
Looks like it should be a good season.
Edit : lots of typos that garbled meaning due to replying on phone.....