improve buttons module