[RB] Updated code to work with gradle