Train dragon updates

Still not working properly