Update ()

- last minute update
- fixes button color background not changing properly