add follow model and controller and fix button