Move the WITH-related semantic check bits to the WITH node

Signed-off-by: Jakub Wieczorek <jakub@jakub.cc>