Return not yet working with assign