Added capitalize function