replace Attack module with calss