Solved Problem through subclassing and using updateLayer method.
I used the same image for highlighted and not highlighted button, but obviously I could leave out the if-statement and use the „self.layer.contents =„ line as the only line in the method