small upgrade

Made button in submit screen have an action