Yup, to add new states you just need to create functions for the enter, exit, and step logic an example for an attack state would be
func attack_emter_logic():
pass
func attack_logic():
func attack_exit_logic():