AX-GADGET: a new code for cosmological simulations of Fuzzy Dark Matter and Axion models