Remove extra code; add grep function