Code clean up on DesignScript side