better model, added buttons