Add support for single line scripts