Hey Vladimir,

Thanks for the comment.

I think it's correct (but might be wrong). Here's a source. What makes you think they should be swapped?

However, I did notice it should be max and not argmax. Will change that now.