Add capitalize function