Got buttons working