minor cleanup on script