Undzėnas, Balys, Lietuva