Title: Enhancements to the STAGS computer code