Removed line class. Changed to a generator function.