Loss function
move button color to beginning
a bunch more on how this works