fix(types): make assign return type more accurate + add Assign type

fix(types): make assign return type more accurate + add Assign type #309