Fix for rocket flame and other improvements