clean up / document firework.py