Add new feature for gain type