make bonus feature better