Method to keep certain columns