Add back lines before returns