Clean up prompt code
Make syntax more consistent with other scripts