Fixed remove_punct function