i think almost any plants you can get from a petshop are suitable. I recommend rooted or potted plants as axies seem to love uprooting plants.. i have great hygrophila, stargrass, banana plants, brazilian pennywort, anubias, and orchid lilly, and baby tears, with java moss starting to grow as well. All of those except for anubias and java moss are fast growing and easy to grow with decent lighting except stargrass which needs a lot.
I'd recommend putting sand under the large stones.