fixed retstmt inside of if, else & while