Everything working but highlighting buttons