Moved gain calculator classes to separate module