Improved return value for shape_outline()
Updated the train and test script for shape:outline