adding refactor of the add columns