Added support for lines. Removed unnecessary function input argument.