Add classes to buttons