Main now takes code on stdin