Put unit attack into own class