Remove redundant code, added docs for stag_columns